package com.example.demo.leetcode.level2;

import java.util.HashSet;
import java.util.Set;

/**
 *
 * 编写一个算法来判断一个数 n 是不是快乐数。
 *
 * 「快乐数」 定义为：
 *
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果这个过程 结果为 1，那么这个数就是快乐数。
 * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：n = 19
 * 输出：true
 * 解释：
 * 1的平方 + 9的平方 = 82
 * 8的平方 + 2的平方 = 68
 * 6的平方 + 8的平方 = 100
 * 1的平方 + 0的平方 + 0的平方 = 1
 * 示例 2：
 *
 * 输入：n = 2
 * 输出：false
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/happy-number
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 */
public class HappyNumber {

    static boolean isHappy(int n) {

        Set set = new HashSet();
        set.add(n);
        int sum = 0;

        while(true){
            while(n > 0){
                int x = n % 10;
                n = n /10;
                sum = sum + x * x;

            }
            if(sum == 1){
                return true;
            }
            n = sum;
            sum = 0;
            if(set.contains(n)){
                return false;
            }
            set.add(n);
        }
    }

    public static void main(String[] args) {

        System.out.println(isHappy(3));

    }

}
